python program to draw a rainbow